import torch
import torchvision
from thop import profile

model = torchvision.models.resnet18(pretrained=False)
input = torch.randn(1, 3, 224, 224)
flop, params = profile(model, inputs=(input,))

print("-------------------------------")
print("flops",flop)
print("-------------------------------")
print("params",params)
